conda activate pytorch
cd /root/mult_gpu/source_ddp
CUDA_VISIBLE_DEVICES=4,5,6,7 OMP_NUM_THREADS=4 python -m torch.distributed.launch --nproc_per_node=4 --use_env train_multi_gpu_using_launch.py

